/* USER CODE BEGIN Header */
/**
 ******************************************************************************
 * @file           : main.c
 * @brief          : Main program body
 ******************************************************************************
 * @attention
 *
 * Copyright (c) 2025 STMicroelectronics.
 * All rights reserved.
 *
 * This software is licensed under terms that can be found in the LICENSE file
 * in the root directory of this software component.
 * If no LICENSE file comes with this software, it is provided AS-IS.
 *
 ******************************************************************************
 */
/* USER CODE END Header */
/* Includes ------------------------------------------------------------------*/
#include "main.h"
#include "dma.h"
#include "app_fatfs.h"
#include "i2s.h"
#include "spi.h"
#include "usart.h"
#include "gpio.h"

/* Private includes ----------------------------------------------------------*/
/* USER CODE BEGIN Includes */
#include "SDcard.h"
#include "i2s_audio_recorder.h"
/* USER CODE END Includes */

/* Private typedef -----------------------------------------------------------*/
/* USER CODE BEGIN PTD */
/* 按键状态定义 */
#define KEY_PRESSED  1 // 按键按下状态
#define KEY_RELEASED 0 // 按键释放状态

/* 录音状态定义 */
#define RECORD_STOPPED 0 // 录音停止状态
#define RECORD_RUNNING 1 // 录音运行状态

/* 全局变量 */
static uint32_t record_start_tick = 0;            // 录音开始时的系统滴答计数
static uint8_t last_key_state     = KEY_RELEASED; // 上一次按键状态记录

/* 函数声明 */
uint8_t get_Key(void);
void Error_Handler(void);

/* 外部引用 */
extern uint16_t audio_buffer[BUFFER_SIZE];
extern volatile uint8_t recording_active;
extern volatile uint8_t buffer_ready;
extern volatile uint8_t current_buffe;
/* USER CODE END PTD */

/* Private define ------------------------------------------------------------*/
/* USER CODE BEGIN PD */

/* USER CODE END PD */

/* Private macro -------------------------------------------------------------*/
/* USER CODE BEGIN PM */

/* USER CODE END PM */

/* Private variables ---------------------------------------------------------*/

/* USER CODE BEGIN PV */

/* USER CODE END PV */

/* Private function prototypes -----------------------------------------------*/
void SystemClock_Config(void);
/* USER CODE BEGIN PFP */

/* USER CODE END PFP */

/* Private user code ---------------------------------------------------------*/
/* USER CODE BEGIN 0 */

/* USER CODE END 0 */

/**
 * @brief  The application entry point.
 * @retval int
 */
int main(void)
{

    /* USER CODE BEGIN 1 */

    /* USER CODE END 1 */

    /* MCU Configuration--------------------------------------------------------*/

    /* Reset of all peripherals, Initializes the Flash interface and the Systick. */
    HAL_Init();

    /* USER CODE BEGIN Init */

    /* USER CODE END Init */

    /* Configure the system clock */
    SystemClock_Config();

    /* USER CODE BEGIN SysInit */

    /* USER CODE END SysInit */

    /* Initialize all configured peripherals */
    MX_GPIO_Init();
    MX_DMA_Init();
    MX_SPI1_Init();
    if (MX_FATFS_Init() != APP_OK) {
        Error_Handler();
    }
    MX_I2S2_Init();
    MX_USART1_UART_Init();
    /* USER CODE BEGIN 2 */
    uint8_t led_tick = 0;
    HAL_Delay(100);
    printf("system start\r\n");
    Get_SDCard_Capacity(); // 得到使用内存并选择格式化
    /* 初始化音频录制系统 */
    if (AudioRecorder_Init() != HAL_OK) {
        printf("Audio system initialization failed!\r\n");
        while (1) LED_R_Toggle(); // 初始化失败时停机
    }
    printf("Audio system initialized. Press key to start recording...\r\n");
    /* USER CODE END 2 */

    /* Infinite loop */
    /* USER CODE BEGIN WHILE */
    while (1) {
        uint8_t current_key = get_Key(); // 获取当前按键状态

        /* 检测按键按下动作（上升沿） */
        if (current_key == KEY_PRESSED && last_key_state == KEY_RELEASED) {
            // 根据当前录音状态执行相应操作
            if (recording_active == RECORD_STOPPED) {
                // 开始录音
                printf("new start\n");
                // if (1) {
                // recording_active  = 1;
                if (StartRecording() == HAL_OK) {
                    record_start_tick = HAL_GetTick(); // 记录开始时间
                    printf("Recording started...\r\n");
                } else {
                    printf("Failed to start recording!\r\n");
                }
            } else {
                // 停止录音
                printf("TT!\r\n");
                StopRecording();
                uint32_t record_duration = HAL_GetTick() - record_start_tick;
                // 计算并打印录音时长（秒.毫秒）
                printf("Recording stopped. Duration: %d.%03d seconds\r\n",
                       (int)(record_duration / 1000),
                       (int)(record_duration % 1000));
            }
        }

        /* 当缓冲区准备好时处理音频数据 */
        if (buffer_ready) {
            printf("p\n");
            ProcessAudioData();
        }
        led_tick++;
        if (recording_active == 1) {
            if (led_tick % 5 == 0) LED_B_Toggle();
        } else {
            if (led_tick % 20 == 0) LED_B_Toggle();
        }
        // last_key_state = current_key; // 更新按键状态记录
        HAL_Delay(10); // 软件消抖延时
        /* USER CODE END WHILE */

        /* USER CODE BEGIN 3 */
    }
    /* USER CODE END 3 */
}

/**
 * @brief System Clock Configuration
 * @retval None
 */
void SystemClock_Config(void)
{
    RCC_OscInitTypeDef RCC_OscInitStruct = {0};
    RCC_ClkInitTypeDef RCC_ClkInitStruct = {0};

    /** Configure the main internal regulator output voltage
     */
    HAL_PWREx_ControlVoltageScaling(PWR_REGULATOR_VOLTAGE_SCALE1_BOOST);

    /** Initializes the RCC Oscillators according to the specified parameters
     * in the RCC_OscInitTypeDef structure.
     */
    RCC_OscInitStruct.OscillatorType = RCC_OSCILLATORTYPE_HSE;
    RCC_OscInitStruct.HSEState       = RCC_HSE_ON;
    RCC_OscInitStruct.PLL.PLLState   = RCC_PLL_ON;
    RCC_OscInitStruct.PLL.PLLSource  = RCC_PLLSOURCE_HSE;
    RCC_OscInitStruct.PLL.PLLM       = RCC_PLLM_DIV1;
    RCC_OscInitStruct.PLL.PLLN       = 40;
    RCC_OscInitStruct.PLL.PLLP       = RCC_PLLP_DIV2;
    RCC_OscInitStruct.PLL.PLLQ       = RCC_PLLQ_DIV2;
    RCC_OscInitStruct.PLL.PLLR       = RCC_PLLR_DIV2;
    if (HAL_RCC_OscConfig(&RCC_OscInitStruct) != HAL_OK) {
        Error_Handler();
    }

    /** Initializes the CPU, AHB and APB buses clocks
     */
    RCC_ClkInitStruct.ClockType      = RCC_CLOCKTYPE_HCLK | RCC_CLOCKTYPE_SYSCLK | RCC_CLOCKTYPE_PCLK1 | RCC_CLOCKTYPE_PCLK2;
    RCC_ClkInitStruct.SYSCLKSource   = RCC_SYSCLKSOURCE_PLLCLK;
    RCC_ClkInitStruct.AHBCLKDivider  = RCC_SYSCLK_DIV1;
    RCC_ClkInitStruct.APB1CLKDivider = RCC_HCLK_DIV1;
    RCC_ClkInitStruct.APB2CLKDivider = RCC_HCLK_DIV1;

    if (HAL_RCC_ClockConfig(&RCC_ClkInitStruct, FLASH_LATENCY_4) != HAL_OK) {
        Error_Handler();
    }
}

/* USER CODE BEGIN 4 */
/* 按键读取函数 */
uint8_t get_Key(void)
{
    // 读取按键状态
    return HAL_GPIO_ReadPin(KEY_GPIO_Port, KEY_Pin);
}

/* USER CODE END 4 */

/**
 * @brief  This function is executed in case of error occurrence.
 * @retval None
 */
void Error_Handler(void)
{
    /* USER CODE BEGIN Error_Handler_Debug */
    /* User can add his own implementation to report the HAL error return state */
    __disable_irq();
    while (1) {
        HAL_Delay(50);
        LED_R_Toggle();
    }
    /* USER CODE END Error_Handler_Debug */
}
#ifdef USE_FULL_ASSERT
/**
 * @brief  Reports the name of the source file and the source line number
 *         where the assert_param error has occurred.
 * @param  file: pointer to the source file name
 * @param  line: assert_param error line source number
 * @retval None
 */
void assert_failed(uint8_t *file, uint32_t line)
{
    /* USER CODE BEGIN 6 */
    /* User can add his own implementation to report the file name and line number,
       ex: printf("Wrong parameters value: file %s on line %d\r\n", file, line) */
    /* USER CODE END 6 */
}
#endif /* USE_FULL_ASSERT */
